package com.clown.networkFlowAnalysis.service;

import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;

/**
 * redis 服务
 *
 * @author clown
 */
public class RedisService {
    /**
     * 服务器IP地址
     */
    private static final String ADDR = "127.0.0.1";
    /**
     * 端口
     */
    private static final int PORT = 6379;
    /**
     * 密码
     */
    private static final String AUTH = "xq990518";
    /**
     * 连接实例的最大连接数
     */
    private static final int MAX_ACTIVE = 1024;
    /**
     * 控制一个pool最多有多少个状态为idle(空闲的)的jedis实例，默认值也是8。
     */
    private static final int MAX_IDLE = 200;
    /**
     * 等待可用连接的最大时间，单位毫秒，默认值为-1，表示永不超时。如果超过等待时间，则直接抛出JedisConnectionException
     */
    private static final int MAX_WAIT = 10000;
    /**
     * 连接超时的时间
     */
    private static final int TIMEOUT = 10000;
    /**
     * 在borrow一个jedis实例时，是否提前进行validate操作；如果为true，则得到的jedis实例均是可用的；
     */
    private static final boolean TEST_ON_BORROW = true;

    private static JedisPool jedisPool = null;
    /**
     * 数据库模式是16个数据库 0~15
     */
    public static final int DEFAULT_DATABASE = 0;

    /**
     * 初始化Redis连接池
     */
    static {
        try {
            JedisPoolConfig config = new JedisPoolConfig();
            config.setMaxTotal(MAX_ACTIVE);
            config.setMaxIdle(MAX_IDLE);
            config.setMaxWaitMillis(MAX_WAIT);
            config.setTestOnBorrow(TEST_ON_BORROW);
            jedisPool = new JedisPool(config, ADDR, PORT, TIMEOUT, AUTH, DEFAULT_DATABASE);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 获取Jedis实例
     */
    public synchronized static Jedis getJedis() {
        Jedis result = null;
        try {

            if (jedisPool != null) {
                Jedis resource = jedisPool.getResource();
                System.out.println("redis--服务正在运行: " + resource.ping());
                result = resource;
            }

        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }

    private <T> T execute(Function<T, Jedis> fun) {
        Jedis jedis = null;
        try {
            // 从连接池中获取到jedis对象
            jedis = jedisPool.getResource();
            return fun.callback(jedis);
        } finally {
            if (null != jedis) {
                // 关闭，检测连接是否有效，有效则放回到连接池中，无效则重置状态
                jedis.close();
            }
        }
    }

    /**
     * 执行set操作，设置数据
     *
     * @param key
     * @param value
     * @return
     */
    public String set(final String key, final String value) {
        return this.execute(new Function<String, Jedis>() {
            public String callback(Jedis e) {
                return e.set(key, value);
            }
        });
    }

    /**
     * 执行set操作，设置数据，并且设置生存时间，单位：秒
     *
     * @param key
     * @param value
     * @param seconds
     * @return
     */
    public String set(final String key, final String value, final Integer seconds) {
        return this.execute(new Function<String, Jedis>() {
            public String callback(Jedis e) {
                String str = e.set(key, value);
                e.expire(key, seconds);
                return str;
            }
        });
    }

    /**
     * 执行get操作，获取数据
     *
     * @param key
     * @return
     */
    public String get(final String key) {
        return this.execute(new Function<String, Jedis>() {
            public String callback(Jedis e) {
                return e.get(key);
            }
        });
    }

    /**
     * 删除一个key
     *
     * @param key
     * @return
     */
    public Long del(final String key) {
        return this.execute(new Function<Long, Jedis>() {
            public Long callback(Jedis e) {
                return e.del(key);
            }
        });
    }

    /**
     * 设置生存时间，单位：秒
     *
     * @param key
     * @param seconds
     * @return
     */
    public Long expire(final String key, final Integer seconds) {
        return this.execute(new Function<Long, Jedis>() {
            public Long callback(Jedis e) {
                return e.expire(key, seconds);
            }
        });
    }
}
